import MySQLdb


class mysql():

    def __init__(self):
        # 打开数据库连接
        self.db = MySQLdb.connect("localhost", "root", "root", "safetools", charset='utf8')
        # 使用cursor()方法获取操作游标
        self.cursor = self.db.cursor()

    def __del__(self):
        # 关闭数据库连接
        self.db.close()

    def select(self, sql):

        try:
            # 执行SQL语句
            self.cursor.execute(sql)
            # 获取所有记录列表
            results = self.cursor.fetchall()
        except:
            print("Error: " + sql)

        return results

    def insert(self, sql):

        try:
            # 执行sql语句
            self.cursor.execute(sql)
            # 提交到数据库执行
            self.db.commit()
        except:
            # 发生错误时回滚
            self.db.rollback()

    def delete(self, sql):

        try:
            # 执行sql语句
            self.cursor.execute(sql)
            # 提交到数据库执行
            self.db.commit()
        except:
            # 发生错误时回滚
            self.db.rollback()
